#! -*- coding: utf-8 -*-
"""
@Author: AI
@Create Time: 20240625
@Info: 模型训练参数
"""

class Args(object):
    bash_path = '/home/data/llm-ner' # 项目路径
    save_dir = '../model/saved_model' # 新模型保存路径
    ner_model_path = '../model/saved_model/model_20240821_211855/' # ner模型路径

    max_seq_len = 512
    batch_size = 24
    learning_rate = 5e-5
    weight_decay = 0.0
    num_train_epochs = 5
    warmup_ratio = 0.06
    device = 'cuda:0'


args = Args()
